package com.castle.fortress.admin.utils;

import io.minio.*;
import io.minio.errors.MinioException;
import io.minio.http.Method;
import io.minio.messages.Bucket;
import io.minio.messages.DeleteError;
import io.minio.messages.DeleteObject;
import io.minio.messages.Item;
import lombok.Getter;
import lombok.Setter;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * MinIO工具类
 * 提供文件上传、下载、删除等功能
 *
 * @author castle
 */
@Slf4j
@Component
public class MinioUtil {

    // 移除 @Value 注解，直接使用实例变量
    private String endpoint = "http://localhost:9000";
    private String accessKey = "minioadmin";
    private String secretKey = "minioadmin";

    @Getter
    @Setter
    private String bucketName = "castle-bucket";

    private MinioClient minioClient;

    /**
     * 初始化 MinIO 客户端
     */
    private void initMinioClient() {
        if (minioClient == null) {
            minioClient = MinioClient.builder()
                    .endpoint(endpoint)
                    .credentials(accessKey, secretKey)
                    .build();
        }
    }

    /**
     * 检查存储桶是否存在
     *
     * @param bucketName 存储桶名称
     * @return 是否存在
     */
    public boolean bucketExists(String bucketName) {
        try {
            initMinioClient();
            return minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build());
        } catch (Exception e) {
            log.error("检查存储桶是否存在发生错误: {}", e.getMessage(), e);
            return false;
        }
    }

    /**
     * 创建存储桶
     *
     * @param bucketName 存储桶名称
     * @return 是否创建成功
     */
    public boolean makeBucket(String bucketName) {
        try {
            initMinioClient();
            if (!bucketExists(bucketName)) {
                minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
                return true;
            }
        } catch (Exception e) {
            log.error("创建存储桶发生错误: {}", e.getMessage(), e);
        }
        return false;
    }

    /**
     * 列出所有存储桶
     *
     * @return 存储桶列表
     */
    public List<String> listBuckets() {
        try {
            initMinioClient();
            List<Bucket> buckets = minioClient.listBuckets();
            return buckets.stream().map(Bucket::name).collect(Collectors.toList());
        } catch (Exception e) {
            log.error("列出存储桶发生错误: {}", e.getMessage(), e);
            return new ArrayList<>();
        }
    }

    /**
     * 上传文件到MinIO
     *
     * @param file 要上传的文件
     * @return 包含文件路径和URL的Map
     */
    public Map<String, String> putObject(MultipartFile file) throws IOException, MinioException {
        return putObject(file, bucketName);
    }

    /**
     * 上传文件到MinIO指定存储桶
     *
     * @param file       要上传的文件
     * @param bucketName 存储桶名称
     * @return 包含文件路径和URL的Map
     */
    public Map<String, String> putObject(MultipartFile file, String bucketName) throws IOException, MinioException {
        // 确保MinioClient已初始化
        initMinioClient();

        // 确保存储桶存在
        if (!bucketExists(bucketName)) {
            makeBucket(bucketName);
        }

        // 生成唯一文件名，避免文件名冲突
        String originalFilename = file.getOriginalFilename();
        // 使用UTF-8编码处理原始文件名，确保中文字符正确显示
        String encodedOriginalFilename = new String(originalFilename != null ? originalFilename.getBytes("ISO-8859-1") : new byte[0], "UTF-8");
        String fileName = UUID.randomUUID().toString().replace("-", "") +
                 (encodedOriginalFilename != null ? encodedOriginalFilename.substring(encodedOriginalFilename.lastIndexOf(".")) : "");

        // 上传文件
        try (InputStream inputStream = file.getInputStream()) {
            minioClient.putObject(
                    PutObjectArgs.builder()
                            .bucket(bucketName)
                            .object(fileName)
                            .stream(inputStream, file.getSize(), -1)
                            .contentType(file.getContentType())
                            .build()
            );
        } catch (Exception e) {
            log.error("上传文件到MinIO发生错误: {}", e.getMessage(), e);
            throw new MinioException(e.getMessage());
        }

        // 构建结果
        Map<String, String> result = new HashMap<>();
        String fileUrl = getFileUrl(bucketName, fileName);
        result.put("filePath", fileName);
        result.put("fileUrl", fileUrl);
        result.put("fileName", fileName);
        result.put("originalFileName", encodedOriginalFilename);
        return result;
    }

    /**
     * 获取文件URL
     *
     * @param bucketName 存储桶名称
     * @param objectName 对象名称
     * @return 文件URL
     */
    public String getFileUrl(String bucketName, String objectName) {
        return String.format("%s/%s/%s", endpoint, bucketName, objectName);
    }

    /**
     * 获取临时访问URL（带过期时间的预签名URL）
     *
     * @param bucketName 存储桶名称
     * @param objectName 对象名称
     * @param expiry     过期时间（秒）
     * @return 预签名URL
     */
    public String getPresignedObjectUrl(String bucketName, String objectName, int expiry) {
        try {
            initMinioClient();
            return minioClient.getPresignedObjectUrl(
                    GetPresignedObjectUrlArgs.builder()
                            .method(Method.GET)
                            .bucket(bucketName)
                            .object(objectName)
                            .expiry(expiry, TimeUnit.SECONDS)
                            .build()
            );
        } catch (Exception e) {
            log.error("获取预签名URL发生错误: {}", e.getMessage(), e);
            return null;
        }
    }

    /**
     * 获取临时访问URL（带过期时间的预签名URL），使用默认桶
     *
     * @param objectName 对象名称
     * @param expiry     过期时间（秒）
     * @return 预签名URL
     */
    public String getPresignedObjectUrl(String objectName, int expiry) {
        return getPresignedObjectUrl(bucketName, objectName, expiry);
    }

    /**
     * 下载文件
     *
     * @param bucketName 存储桶名称
     * @param objectName 对象名称
     * @return 文件字节数组
     */
    public byte[] getObject(String bucketName, String objectName) {
        try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
            initMinioClient();
            GetObjectResponse response = minioClient.getObject(
                    GetObjectArgs.builder()
                            .bucket(bucketName)
                            .object(objectName)
                            .build()
            );
            IOUtils.copy(response, outputStream);
            return outputStream.toByteArray();
        } catch (Exception e) {
            log.error("下载文件发生错误: {}", e.getMessage(), e);
            return null;
        }
    }

    /**
     * 使用默认桶下载文件
     *
     * @param objectName 对象名称
     * @return 文件字节数组
     */
    public byte[] getObject(String objectName) {
        return getObject(bucketName, objectName);
    }

    /**
     * 下载文件并作为HTTP响应返回
     *
     * @param bucketName  存储桶名称
     * @param objectName  对象名称
     * @param downloadFilename 下载显示的文件名
     * @return ResponseEntity
     */
    public ResponseEntity<byte[]> downloadObject(String bucketName, String objectName, String downloadFilename) {
        try {
            byte[] content = getObject(bucketName, objectName);
            if (content == null) {
                return ResponseEntity.notFound().build();
            }

            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
            headers.setContentDispositionFormData("attachment", downloadFilename);
            headers.setContentLength(content.length);

            return new ResponseEntity<>(content, headers, HttpStatus.OK);
        } catch (Exception e) {
            log.error("下载文件发生错误: {}", e.getMessage(), e);
            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build();
        }
    }

    /**
     * 删除文件
     */
    public boolean removeObject(String fileName) throws Exception {
        initMinioClient();
        minioClient.removeObject(
                RemoveObjectArgs.builder()
                        .bucket(bucketName)
                        .object(fileName)
                        .build()
        );
        return true;
    }

    /**
     * 批量删除文件
     *
     * @param bucketName  存储桶名称
     * @param objectNames 对象名称列表
     * @return 删除结果，包含成功删除的对象和失败的错误信息
     */
    public Map<String, List<String>> removeObjects(String bucketName, List<String> objectNames) {
        List<DeleteObject> deleteObjects = objectNames.stream()
                .map(DeleteObject::new)
                .collect(Collectors.toList());

        Map<String, List<String>> result = new HashMap<>();
        result.put("success", new ArrayList<>());
        result.put("error", new ArrayList<>());

        try {
            Iterable<Result<DeleteError>> results = minioClient.removeObjects(
                    RemoveObjectsArgs.builder()
                            .bucket(bucketName)
                            .objects(deleteObjects)
                            .build()
            );

            // 处理删除结果
            for (Result<DeleteError> r : results) {
                try {
                    DeleteError error = r.get();
                    result.get("error").add(error.objectName() + ": " + error.message());
                } catch (Exception e) {
                    // 如果没有错误获取到，说明删除成功
                }
            }

            // 所有成功删除的对象
            objectNames.removeAll(result.get("error"));
            result.get("success").addAll(objectNames);

            return result;
        } catch (Exception e) {
            log.error("批量删除文件发生错误: {}", e.getMessage(), e);
            result.get("error").add("批量删除异常: " + e.getMessage());
            return result;
        }
    }

    /**
     * 使用默认桶批量删除文件
     *
     * @param objectNames 对象名称列表
     * @return 删除结果
     */
    public Map<String, List<String>> removeObjects(List<String> objectNames) {
        return removeObjects(bucketName, objectNames);
    }

    /**
     * 列出存储桶中的所有对象
     *
     * @param bucketName 存储桶名称
     * @return 对象名称列表
     */
    public List<String> listObjects(String bucketName) {
        try {
            initMinioClient();
            List<String> objectNames = new ArrayList<>();
            Iterable<Result<Item>> results = minioClient.listObjects(
                    ListObjectsArgs.builder()
                            .bucket(bucketName)
                            .build()
            );

            for (Result<Item> result : results) {
                objectNames.add(result.get().objectName());
            }

            return objectNames;
        } catch (Exception e) {
            log.error("列出对象发生错误: {}", e.getMessage(), e);
            return new ArrayList<>();
        }
    }

    /**
     * 使用默认桶列出所有对象
     *
     * @return 对象名称列表
     */
    public List<String> listObjects() {
        return listObjects(bucketName);
    }

    /**
     * 检查对象是否存在
     *
     * @param bucketName 存储桶名称
     * @param objectName 对象名称
     * @return 是否存在
     */
    public boolean objectExists(String bucketName, String objectName) {
        try {
            initMinioClient();
            minioClient.statObject(
                    StatObjectArgs.builder()
                            .bucket(bucketName)
                            .object(objectName)
                            .build()
            );
            return true;
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 使用默认桶检查对象是否存在
     *
     * @param objectName 对象名称
     * @return 是否存在
     */
    public boolean objectExists(String objectName) {
        return objectExists(bucketName, objectName);
    }

    /**
     * 创建文本文件
     *
     * @param bucketName 存储桶名称
     * @param objectName 对象名称
     * @param content    文本内容
     * @return 是否创建成功
     */
    public boolean createTextFile(String bucketName, String objectName, String content) {
        try {
            initMinioClient();
            byte[] contentBytes = content.getBytes();
            ByteArrayInputStream bais = new ByteArrayInputStream(contentBytes);

            minioClient.putObject(
                    PutObjectArgs.builder()
                            .bucket(bucketName)
                            .object(objectName)
                            .stream(bais, contentBytes.length, -1)
                            .contentType(MediaType.TEXT_PLAIN_VALUE)
                            .build()
            );

            return true;
        } catch (Exception e) {
            log.error("创建文本文件发生错误: {}", e.getMessage(), e);
            return false;
        }
    }

    /**
     * 使用默认桶创建文本文件
     *
     * @param objectName 对象名称
     * @param content    文本内容
     * @return 是否创建成功
     */
    public boolean createTextFile(String objectName, String content) {
        return createTextFile(bucketName, objectName, content);
    }

    /**
     * 设置访问凭证和端点
     *
     * @param endpoint 端点URL
     * @param accessKey 访问密钥
     * @param secretKey 密钥
     */
    public void setCredentials(String endpoint, String accessKey, String secretKey) {
        this.endpoint = endpoint;
        this.accessKey = accessKey;
        this.secretKey = secretKey;
        // 重置连接，以便下次使用时使用新的凭证
        this.minioClient = null;
    }

    /**
     * 获取 MinioClient 实例
     *
     * @return MinioClient 实例
     */
    public MinioClient getMinioClient() {
        initMinioClient();
        return minioClient;
    }
}
